how to code linear search python